#!/usr/bin/env python3
"""
基本测试脚本 - 使用正确的模块名
"""

import sys
import os

# 添加项目根目录到Python路径
project_root = os.path.dirname(os.path.abspath(__file__))
sys.path.insert(0, project_root)

print(f"项目根目录: {project_root}")

# 创建符号链接的路径（如果不存在）
wenyan_plus_pp_path = os.path.join(os.path.dirname(project_root), 'wenyan_plus_pp')
if not os.path.exists(wenyan_plus_pp_path):
    try:
        os.symlink(project_root, wenyan_plus_pp_path)
        print(f"已创建符号链接: {wenyan_plus_pp_path} -> {project_root}")
    except:
        print(f"无法创建符号链接，将直接使用项目路径")

# 测试导入核心模块
try:
    from core.hybrid_engine import QuantumHybridEngine
    print("✓ 成功导入 QuantumHybridEngine")
except ImportError as e:
    print(f"✗ 导入 QuantumHybridEngine 失败: {e}")

# 测试导入算法模块
try:
    from algorithms.vqe import VQESolver
    print("✓ 成功导入 VQESolver")
except ImportError as e:
    print(f"✗ 导入 VQESolver 失败: {e}")

# 测试导入量子模块
try:
    from modules.quantum.main import Quantum
    print("✓ 成功导入 Quantum")
    
    # 测试创建实例
    q = Quantum()
    print("✓ 成功创建 Quantum 实例")
    
    # 测试方法调用
    ham = q.获取示例哈密顿量('氢分子')
    print(f"✓ 成功调用获取示例哈密顿量: {ham}")
    
except ImportError as e:
    print(f"✗ 导入 Quantum 失败: {e}")
except Exception as e:
    print(f"✗ 其他错误: {e}")

print("=== 测试完成 ===")
